package sjp.demo.algorithm.binarysearch;

import org.junit.Assert;
import org.junit.Test;

/**
 * https://leetcode-cn.com/problems/search-insert-position/
 * 搜索插入位置
 *
 *
 * 给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。
 *
 * 请必须使用时间复杂度为 O(log n) 的算法。
 *
 *  
 *
 * 示例 1:
 *
 * 输入: nums = [1,3,5,6], target = 5
 * 输出: 2
 * 示例 2:
 *
 * 输入: nums = [1,3,5,6], target = 2
 * 输出: 1
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/search-insert-position
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class SearchInsertPosition {
    public int searchInsert(int[] nums, int target) {
        if (nums.length == 1) {
            return nums[0] < target ? 1 : 0;
        }
        int a = 0;
        int b = nums.length - 1;
        while (a <= b) {
            int i = a + ((b - a) / 2);
            if (nums[i] == target) {
                return i;
            } else if (nums[i] > target) {
                b = i - 1;
            } else {
                a = i + 1;
            }
        }
        if (a > nums.length - 1) {
            a = nums.length - 1;
        }
        return target < nums[a] ? a : a + 1;
    }

    @Test
    public void test() {
        int[] nums = new int[]{1, 3, 5, 6};
        int result = searchInsert(nums, 7);
        Assert.assertEquals(4, result);
    }
}
